import matplotlib.pyplot as plt
import numpy as np

# 中文乱码处理
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
plt.rcParams['figure.dpi'] = 256
plt.rcParams['figure.figsize'] = (5,3)
# 字号调整
plt.rcParams['font.size'] = 6
# 增加图题
# plt.title('轨道吊作业状态对比图')
# plt.title('正面吊作业状态对比图')
plt.title('轨道吊作业状态对比图')
# 绘制对比条形图
def draw_bar_chart():
    # 数据
    x_ = ['等待', '空载移动', '重载移动', '操作集装箱', '避让']
    x = np.arange(len(x_))
    # 轨道吊参数
    y1 = [0, 639, 332, 1540, 0]
    y2 = [17, 427, 328, 1388, 0]
    # 正面吊参数
    # y1 = [1085, 369, 188, 491, 0]
    # y2 = [805, 358, 187, 540, 25]
    # 汇总参数
    # y1 = [1085, 1008, 520, 2031, 0]
    # y2 = [822, 785, 515, 1928, 25]
    width = 0.35

    # 绘制条形图
    plt.bar(x, y1,width = width,label='分区方案',align='center',zorder=1)
    plt.bar(x+width, y2,width = width,label='协同方案',align='center',zorder=2)
    plt.xticks(x + width/2, x_)
    # 添加图例
    plt.legend()
    # 加入y坐标轴标签
    plt.ylabel('持续时间（秒）')

    # 显示图形
    plt.show()

if __name__ == '__main__':
    draw_bar_chart()